import torch

# 创建一个tensor对象, 类似于 numpy
a = torch.tensor([1, 2, 3], dtype=int)
print(a)
# 查看对象的类型 dtype
print(a.dtype)
b = torch.tensor([4, 5, 6], dtype=float)
print(b)

# 调用ndimension()方法来查看 tensor 的维度
tensor = torch.tensor([[1, 2, 3],
                       [3, 4, 5]])
print(tensor.ndimension())

# 调用 shape 或者 size() 来查看数据的形状(就是行和列)
print(tensor.shape)

# print(tensor.size())
